package com.yst.lambda;

import java.io.Serializable;
import java.lang.invoke.SerializedLambda;
import java.lang.reflect.Method;
import java.util.Locale;
import java.util.Map;
import java.util.Objects;
import java.util.concurrent.ConcurrentHashMap;

/**
 * 此辅助单元只为减小对其他第三方包的依赖
 * 具体详细的功能，推荐使用hutool 工具包
 */
public final class LambdaFuncUtils {
    private static final Map<String, SerializedLambda> cache = new ConcurrentHashMap<>();

    private static Method getWriteReplaceMethod(Object obj) throws SecurityException {
        if (null == obj) {
            return null;
        }

        Method[] methods=obj.getClass().getDeclaredMethods();
        for (Method method:methods){
            if (method.getName().equals("writeReplace")){
                return method;
            }
        }
        return null;
    }


    private static SerializedLambda getWriteReplace(Object object){
        final Method method = getWriteReplaceMethod(object);
        if (Objects.nonNull(method)) {
            try {
                method.setAccessible(true);
                return (SerializedLambda) method.invoke(object);
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
        return null;
    }

    private static String methodNameToFieldName(String methodName){
        if (methodName.length()<4) return methodName;
        return methodName.substring(3,4).toLowerCase(Locale.ENGLISH).concat(methodName.substring(4));
    }

    /**
     * 解析lambda表达式,加了缓存。
     * 该缓存可能会在任意不定的时间被清除。
     *
     * <p>
     * 通过反射调用实现序列化接口函数对象的writeReplace方法，从而拿到{@link SerializedLambda}<br>
     * 该对象中包含了lambda表达式的所有信息。
     * </p>
     *
     * @param func 需要解析的 lambda 对象
     * @return 返回解析后的结果
     */
    private static SerializedLambda resolve(Serializable func) {
        return cache.computeIfAbsent(func.getClass().getName(),
                (key) -> getWriteReplace(func));
    }

    /**
     * 获取函数的名称
     * @param tableFieldFunc 函数
     * @return 函数名称
     */
    public static String getFuncName(TableFieldFunc<?,?> tableFieldFunc){
        SerializedLambda serializedLambda=resolve(tableFieldFunc);
        if (Objects.nonNull(serializedLambda)) {
            return resolve(tableFieldFunc).getImplMethodName();
        }
        else {
            return null;
        }
    }

    /**
     * 获取函数所局实体类名称
     * @param tableFieldFunc 函数
     * @return
     */
    public static String getOwnerClassName(TableFieldFunc<?,?> tableFieldFunc){
        SerializedLambda serializedLambda=resolve(tableFieldFunc);
        if (Objects.nonNull(serializedLambda)) {
            return resolve(tableFieldFunc).getImplClass();
        }
        else {
            return null;
        }
    }

    public static Class<?> getOwnerClass(TableFieldFunc<?,?> tableFieldFunc){
        SerializedLambda serializedLambda=resolve(tableFieldFunc);
        if (Objects.nonNull(serializedLambda)) {
            String className= resolve(tableFieldFunc).getImplClass();
            try {
                return Class.forName(className);
            }
            catch (Exception ignore){
                return null;
            }
        }
        else {
            return null;
        }
    }

    public static String getOwnerSimpleClassName(TableFieldFunc<?,?> tableFieldFunc){
        SerializedLambda serializedLambda=resolve(tableFieldFunc);
        if (Objects.nonNull(serializedLambda)) {
            String className= resolve(tableFieldFunc).getImplClass();
            if (className.contains("$")){
                return className.substring(className.indexOf("$")+1);
            }
            return className;
        }
        else {
            return null;
        }
    }

    /**
     * 由函数获取字段名
     * @param tableFieldFunc 函数
     * @return 字段名
     */
    public static String getFieldName(TableFieldFunc<?,?> tableFieldFunc){
        String strFuncName=getFuncName(tableFieldFunc);
        if (Objects.isNull(strFuncName)) return null;
        return methodNameToFieldName(strFuncName);
    }

}
